package com.wechat.utils;

import com.swak.utils.ChineseUtils;
import com.swak.utils.StringUtils;

/**
 * 拼音转换工具
 * 代码描述
 * @author syd
 * @Copyright: Copyright (C) 2021-2028, Inc. All rights reserved.
 * @date Date : 2022-02-11 20:52
 */
import net.sourceforge.pinyin4j.PinyinHelper;
import net.sourceforge.pinyin4j.format.HanyuPinyinCaseType;
import net.sourceforge.pinyin4j.format.HanyuPinyinOutputFormat;
import net.sourceforge.pinyin4j.format.HanyuPinyinToneType;
import net.sourceforge.pinyin4j.format.exception.BadHanyuPinyinOutputFormatCombination;

public class PinYin4jUtils {

	/**
	 * 将字符串转换成拼音数组
	 *
	 * @param src
	 * @return
	 */
	public static String pinyinByString(String src) {
		String[] res = pinyinByString(src, false, null);
		return res != null && res.length > 0 ? StringUtils.join(res, StringUtils.EMPTY) : StringUtils.EMPTY;
	}

	/**
	 * 将字符串转换成拼音数组
	 *
	 * @param src
	 * @return
	 */
	public static String pinyinByString(String src, String separator) {
		String[] res = pinyinByString(src, false, separator);
		return res != null && res.length > 0 ? StringUtils.join(res, StringUtils.EMPTY) : StringUtils.EMPTY;
	}

	/**
	 * 将字符串转换成拼音数组
	 *
	 * @param src
	 * @param isPolyphone 是否查出多音字的所有拼音
	 * @param separator 多音字拼音之间的分隔符
	 * @return
	 */
	public static String[] pinyinByString(String src, boolean isPolyphone, String separator) {
		// 判断字符串是否为空
		if (StringUtils.EMPTY.equals(src) || null == src) {
			return null;
		}
		char[] srcChar = src.toCharArray();
		int srcCount = srcChar.length;
		String[] srcStr = new String[srcCount];

		for (int i = 0; i < srcCount; i++) {
			srcStr[i] = charToPinyin(srcChar[i], isPolyphone, separator);
		}
		return srcStr;
	}

	/**
	 * 将单个字符转换成拼音
	 *
	 * @param src
	 * @return
	 */
	private static String charToPinyin(char src, boolean isPolyphone, String separator) {
		// 创建汉语拼音处理类
		HanyuPinyinOutputFormat defaultFormat = new HanyuPinyinOutputFormat();
		// 输出设置，大小写，音标方式
		defaultFormat.setCaseType(HanyuPinyinCaseType.LOWERCASE);
		defaultFormat.setToneType(HanyuPinyinToneType.WITHOUT_TONE);

		StringBuffer tempPinying = new StringBuffer();

		// 如果是中文
		if (ChineseUtils.isChinese(src)) {
			try {
				// 转换得出结果
				String[] strs = PinyinHelper.toHanyuPinyinStringArray(src, defaultFormat);

				// 是否查出多音字，默认是查出多音字的第一个字符
				if (isPolyphone && null != separator) {
					for (int i = 0; i < strs.length; i++) {
						tempPinying.append(strs[i]);
						if (strs.length != (i + 1)) {
							// 多音字之间用特殊符号间隔起来
							tempPinying.append(separator);
						}
					}
				} else if (strs.length > 0) {
					tempPinying.append(strs[0]);
				} else {
					tempPinying.append(src);
				}

			} catch (BadHanyuPinyinOutputFormatCombination e) {
			}
		} else {
			tempPinying.append(src);
		}
		return tempPinying.toString();

	}

//	/**
//	 * ratain true保留不能被转换为拼音的字符比如说数字，false舍弃
//	 * @param hanzi
//	 * @param retain
//	 * @return
//	 */
//	public static String hanziToPinyin(String hanzi, boolean retain) {
//		return hanziToPinyin(hanzi, "", retain);
//	}
//
//	/**
//	 * 将汉字转换成拼音
//	 *
//	 * @param hanzi
//	 * @param separator
//	 * @return
//	 */
//	public static String hanziToPinyin(String hanzi, String separator, boolean retain) {
//
//		// 创建汉语拼音处理类
//		HanyuPinyinOutputFormat defaultFormat = new HanyuPinyinOutputFormat();
//		// 输出设置，大小写，音标方式
//		defaultFormat.setCaseType(HanyuPinyinCaseType.LOWERCASE);
//		defaultFormat.setToneType(HanyuPinyinToneType.WITHOUT_TONE);
//
//		String pinyingStr = "";
//		try {
//			pinyingStr = PinyinHelper.toHanYuPinyinString(hanzi, defaultFormat, separator, retain);
//		} catch (BadHanyuPinyinOutputFormatCombination e) {
//			// ig
//		}
//		return pinyingStr;
//	}

	/**
	 * 取汉字的首字母
	 *
	 * @param src
	 * @param isCapital 是否是大写
	 * @return
	 */
	public static char[] headByChar(char src, boolean isCapital) {
		// 如果不是汉字直接返回
		if (!ChineseUtils.isChinese(src)) {
			return new char[] { src };
		}
		// 获取所有的拼音
		String[] pinyingStr = PinyinHelper.toHanyuPinyinStringArray(src);
		if (pinyingStr == null) {
			return new char[] { src };
		}

		// 创建返回对象
		int polyphoneSize = pinyingStr.length;
		char[] headChars = new char[polyphoneSize];
		int i = 0;
		// 截取首字符
		for (String s : pinyingStr) {
			char headChar = s.charAt(0);
			// 首字母是否大写，默认是小写
			if (isCapital) {
				headChars[i] = Character.toUpperCase(headChar);
			} else {
				headChars[i] = headChar;
			}
			i++;
		}

		return headChars;
	}

	/**
	 * 取汉字的首字母(默认是大写)
	 *
	 * @param src
	 * @return
	 */
	public static char[] headByChar(char src) {
		return headByChar(src, true);
	}

	/**
	 * 查找字符串首字母
	 *
	 * @param src
	 * @return
	 */
	public static String headByString(String src) {
		return headByString(src, true);
	}

	/**
	 * 查找字符串首字母
	 *
	 * @param src
	 * @param isCapital
	 *            是否大写
	 * @return
	 */
	public static String headByString(String src, boolean isCapital) {
		return headByString(src, isCapital, null);
	}

	/**
	 * 查找字符串首字母
	 *
	 * @param src
	 * @param isCapital 是否大写
	 * @param separator 分隔符
	 * @return
	 */
	public static String headByString(String src, boolean isCapital, String separator) {
		char[] chars = src.toCharArray();
		String[] headString = new String[chars.length];
		int i = 0;
		for (char ch : chars) {
			char[] chs = headByChar(ch, isCapital);
			StringBuffer sb = new StringBuffer();
			if (null != separator) {
				int j = 1;

				for (char ch1 : chs) {
					sb.append(ch1);
					if (j != chs.length) {
						sb.append(separator);
					}
					j++;
				}
			} else if (chs.length > 0) {
				sb.append(chs[0]);
			}
			headString[i] = sb.toString();
			i++;
		}
		return StringUtils.join(headString, StringUtils.EMPTY);
	}
}
